package com.librelato.tela.datamodel;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import com.librelato.bean.BemAtivoBean;

public class BemDataModel extends ListDataModel<BemAtivoBean>
		implements SelectableDataModel<BemAtivoBean> {

	public BemDataModel(List<BemAtivoBean> data) {
		super(data);
	}

	@SuppressWarnings("unchecked")
	@Override
	public BemAtivoBean getRowData(String linha) {
		List<BemAtivoBean> listaBem = (List<BemAtivoBean>) getWrappedData();
		for (BemAtivoBean bem : listaBem) {
			if (bem.getCodigo().equals(linha))
				return bem;

		}

		return null;
	}

	@Override
	public Object getRowKey(BemAtivoBean linha) {
		return linha.getCodigo();
	}

}
